home *** CD-ROM | disk | FTP | other *** search
/ Personal Computer World 2009 February / PCWFEB09.iso / Software / Linux / Kubuntu 8.10 / kubuntu-8.10-desktop-i386.iso / casper / filesystem.squashfs / usr / share / pyshared / PIL / XpmImagePlugin.py < prev    next >
Text File  |  2006-12-03  |  3KB  |  130 lines
  1. #
  2. # The Python Imaging Library.
  3. # $Id: XpmImagePlugin.py 2134 2004-10-06 08:55:20Z fredrik $
  4. #
  5. # XPM File handling
  6. #
  7. # History:
  8. # 1996-12-29 fl   Created
  9. # 2001-02-17 fl   Use 're' instead of 'regex' (Python 2.1) (0.7)
  10. #
  11. # Copyright (c) Secret Labs AB 1997-2001.
  12. # Copyright (c) Fredrik Lundh 1996-2001.
  13. #
  14. # See the README file for information on usage and redistribution.
  15. #
  16.  
  17.  
  18. __version__ = "0.2"
  19.  
  20.  
  21. import re, string
  22. import Image, ImageFile, ImagePalette
  23.  
  24. # XPM header
  25. xpm_head = re.compile("\"([0-9]*) ([0-9]*) ([0-9]*) ([0-9]*)")
  26.  
  27.  
  28. def _accept(prefix):
  29.     return prefix[:9] == "/* XPM */"
  30.  
  31. ##
  32. # Image plugin for X11 pixel maps.
  33.  
  34. class XpmImageFile(ImageFile.ImageFile):
  35.  
  36.     format = "XPM"
  37.     format_description = "X11 Pixel Map"
  38.  
  39.     def _open(self):
  40.  
  41.         if not _accept(self.fp.read(9)):
  42.             raise SyntaxError, "not an XPM file"
  43.  
  44.         # skip forward to next string
  45.         while 1:
  46.             s = self.fp.readline()
  47.             if not s:
  48.                 raise SyntaxError, "broken XPM file"
  49.             m = xpm_head.match(s)
  50.             if m:
  51.                 break
  52.  
  53.         self.size = int(m.group(1)), int(m.group(2))
  54.  
  55.         pal = int(m.group(3))
  56.         bpp = int(m.group(4))
  57.  
  58.         if pal > 256 or bpp != 1:
  59.             raise ValueError, "cannot read this XPM file"
  60.  
  61.         #
  62.         # load palette description
  63.  
  64.         palette = ["\0\0\0"] * 256
  65.  
  66.         for i in range(pal):
  67.  
  68.             s = self.fp.readline()
  69.             if s[-2:] == '\r\n':
  70.                 s = s[:-2]
  71.             elif s[-1:] in '\r\n':
  72.                 s = s[:-1]
  73.  
  74.             c = ord(s[1])
  75.             s = string.split(s[2:-2])
  76.  
  77.             for i in range(0, len(s), 2):
  78.  
  79.                 if s[i] == "c":
  80.  
  81.                     # process colour key
  82.                     rgb = s[i+1]
  83.                     if rgb == "None":
  84.                         self.info["transparency"] = c
  85.                     elif rgb[0] == "#":
  86.                         # FIXME: handle colour names (see ImagePalette.py)
  87.                         rgb = string.atoi(rgb[1:], 16)
  88.                         palette[c] = chr((rgb >> 16) & 255) +\
  89.                                      chr((rgb >> 8) & 255) +\
  90.                                      chr(rgb & 255)
  91.                     else:
  92.                         # unknown colour
  93.                         raise ValueError, "cannot read this XPM file"
  94.                     break
  95.  
  96.             else:
  97.  
  98.                 # missing colour key
  99.                 raise ValueError, "cannot read this XPM file"
  100.  
  101.         self.mode = "P"
  102.         self.palette = ImagePalette.raw("RGB", string.join(palette, ""))
  103.  
  104.         self.tile = [("raw", (0, 0)+self.size, self.fp.tell(), ("P", 0, 1))]
  105.  
  106.     def load_read(self, bytes):
  107.  
  108.         #
  109.         # load all image data in one chunk
  110.  
  111.         xsize, ysize = self.size
  112.  
  113.         s = [None] * ysize
  114.  
  115.         for i in range(ysize):
  116.             s[i] = string.ljust(self.fp.readline()[1:xsize+1], xsize)
  117.  
  118.         self.fp = None
  119.  
  120.         return string.join(s, "")
  121.  
  122. #
  123. # Registry
  124.  
  125. Image.register_open("XPM", XpmImageFile, _accept)
  126.  
  127. Image.register_extension("XPM", ".xpm")
  128.  
  129. Image.register_mime("XPM", "image/xpm")
  130.